// 54.螺旋矩阵
package LeetcodeHot100;

import java.util.ArrayList;
import java.util.HashMap;
import java.util.List;
import java.util.Map;

public class Solution054 {
    public List<Integer> spiralOrder(int[][] matrix) {
        List<Integer> result = new ArrayList<>();
        // 记录四种方向坐标的变化形式
        Map<Integer, int[]> map = new HashMap<>();
        map.put(0, new int[] { 0, 1 });
        map.put(1, new int[] { 1, 0 });
        map.put(2, new int[] { 0, -1 });
        map.put(3, new int[] { -1, 0 });
        int[] location = new int[] { 0, 0 }; // 当前的位置坐标
        int direction = 0; // 当前的方向
        int count = matrix.length * matrix[0].length; // 元素的总个数
        while (count-- != 0) {
            result.add(matrix[location[0]][location[1]]); // 将当前坐标的元素加入结果
            matrix[location[0]][location[1]] = Integer.MIN_VALUE; // 设置当前元素已经被访问
            // 尝试根据当前方向更新坐标，判断是否能更新，如果不能则改变方向
            int i = location[0] + map.get(direction % 4)[0];
            int j = location[1] + map.get(direction % 4)[1];
            if (i < 0 || i > matrix.length - 1 || j < 0 || j > matrix[0].length - 1
                    || matrix[i][j] == Integer.MIN_VALUE)
                direction++;
            location[0] = location[0] + map.get(direction % 4)[0];
            location[1] = location[1] + map.get(direction % 4)[1];
        }
        return result;
    }
}
